from langchain_core.messages import HumanMessage, AIMessage

from src.custom_chat_model_class.parrot_link import ChatParrotLink

model = ChatParrotLink(parrot_buffer_length=3, model="my_custom_model")

result = model.invoke(
    [
        HumanMessage(content="hello!"),
        HumanMessage(content="Meow!"),
    ]
)
print(result)